how to write python code connect database